import pandas as pd
import matplotlib.pyplot as plt

# 设置中文字体为SimHei，解决中文乱码问题
plt.rcParams['font.sans-serif'] = ['SimHei']
plt.rcParams['axes.unicode_minus'] = False

# 读取CSV文件，使用gbk编码
df = pd.read_csv('data.csv', encoding='gbk')

# 提取数据
data_size = df['数据规模']
bruteforce_time = df['蛮力法']
divide_conquer_time = df['分治法']

# 创建画布
fig, ax = plt.subplots()



# 绘制蛮力法和分治法的折线图，并使用不同符号
ax.set_xlabel('数据规模')
ax.set_ylabel('时间')
ax.plot(data_size, bruteforce_time, marker='o', label='蛮力法', color='red')
ax.plot(data_size, divide_conquer_time, marker='s', label='分治法', color='green')

# 显示网格
plt.grid(True)

# 显示图例
fig.tight_layout()
fig.legend(loc='center right')

# 显示图形
plt.title('蛮力法和分治法时间变化')
plt.savefig('蛮力法和分治法时间变化-共享Y轴.png', dpi=300)
plt.show()
